# encoding: utf-8

import matplotlib.pyplot as plt
import numpy as np
from sklearn import svm
from sklearn import linear_model
from sklearn.preprocessing import PolynomialFeatures


def train_multi(X:np.ndarray, Y:np.ndarray):
    poly_reg = PolynomialFeatures(degree=1)
    X_ploy = poly_reg.fit_transform(X)
    trainer = linear_model.LinearRegression()
    trainer.fit(X_ploy, Y)
    return trainer, poly_reg